package cn.junki.sparkbigmodeldemo.entity.dto;

import lombok.*;

import java.io.Serializable;

/**
 * 对话消息封装
 *
 * @author Junki
 * @date 2023-09-21 12:51:12
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
@Builder
public class SparkBigModelChatMessage implements Serializable {

    /**
     * 角色
     */
    private String role;

    /**
     * 内容
     */
    private String content;

    /**
     * 角色枚举
     */
    @Getter
    @AllArgsConstructor
    enum ChatMessageRoleEnum {
        USER("user"),
        AI("assistant");

        final String code;
    }

}
